package com.ruoyi.ade.llm.model;

import java.util.List;

/**
 * 嵌入模型
 */
public class EmbeddingRequest {

    public static final String MODEL_DMETS ="shaw/dmeta-embedding-zh:latest";  //400M参数
    public static final String MODEL_NOMIC = "nomic-embed-text:latest";  //137M参数
    public static final String MODEL_BGE = "bge-m3:567m";  //

    public EmbeddingRequest(String model, List<String> inputs){
        this.model = model;
        this.input = inputs;
    }

    private String model;
    private List<String> input;

    public List<String> getInput() {
        return input;
    }

    public void setInput(List<String> input) {
        this.input = input;
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }
}
